Premchandyadav369 commited on
Commit
4bc8b56
·
1 Parent(s): ff19116

fix(zerogpu): import spaces as first module before torch

Browse files
Files changed (1) hide show
  1. app.py +10 -8
app.py CHANGED
@@ -1,5 +1,5 @@
1
  #!/usr/bin/env python3
2
- """
3
  Hugging Face Spaces Interactive Showcase: Q-TensorFormer.
4
 
5
  Multi-Tab Frontier Systems Hub:
@@ -15,12 +15,7 @@ Multi-Tab Frontier Systems Hub:
15
  - 14 architectures, 17 metrics, and mathematical formalisms
16
  """
17
 
18
- import os
19
- import time
20
- import torch
21
- from configuration_qtensorformer import QTensorFormerConfig
22
- from modeling_qtensorformer import QTensorFormerForCausalLM
23
-
24
  try:
25
  import spaces
26
  HAS_SPACES = True
@@ -33,6 +28,14 @@ def gpu_decorator(fn):
33
  return spaces.GPU(fn)
34
  return fn
35
 
 
 
 
 
 
 
 
 
36
  try:
37
  import gradio as gr
38
  HAS_GRADIO = True
@@ -40,7 +43,6 @@ except ImportError:
40
  HAS_GRADIO = False
41
 
42
 
43
- from pathlib import Path
44
  from src.tokenizer import QTensorFormerTokenizer
45
 
46
  # Initialize deterministic tokenizer
 
1
  #!/usr/bin/env python3
2
+ r"""
3
  Hugging Face Spaces Interactive Showcase: Q-TensorFormer.
4
 
5
  Multi-Tab Frontier Systems Hub:
 
15
  - 14 architectures, 17 metrics, and mathematical formalisms
16
  """
17
 
18
+ # IMPORTANT: On Hugging Face ZeroGPU (zero-a10g), `spaces` MUST be imported before `torch`
 
 
 
 
 
19
  try:
20
  import spaces
21
  HAS_SPACES = True
 
28
  return spaces.GPU(fn)
29
  return fn
30
 
31
+
32
+ import os
33
+ import time
34
+ from pathlib import Path
35
+ import torch
36
+ from configuration_qtensorformer import QTensorFormerConfig
37
+ from modeling_qtensorformer import QTensorFormerForCausalLM
38
+
39
  try:
40
  import gradio as gr
41
  HAS_GRADIO = True
 
43
  HAS_GRADIO = False
44
 
45
 
 
46
  from src.tokenizer import QTensorFormerTokenizer
47
 
48
  # Initialize deterministic tokenizer