Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import os
|
|
3 |
import time
|
4 |
from os import path
|
5 |
from safetensors.torch import load_file
|
6 |
-
|
7 |
import spaces
|
8 |
|
9 |
cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
|
@@ -16,8 +16,7 @@ import torch
|
|
16 |
from diffusers import FluxPipeline
|
17 |
|
18 |
torch.backends.cuda.matmul.allow_tf32 = True
|
19 |
-
|
20 |
-
huggingface_hub.login(os.getenv('HF_TOKEN'))
|
21 |
class timer:
|
22 |
def __init__(self, method_name="timed process"):
|
23 |
self.method = method_name
|
|
|
3 |
import time
|
4 |
from os import path
|
5 |
from safetensors.torch import load_file
|
6 |
+
import huggingface_hub
|
7 |
import spaces
|
8 |
|
9 |
cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
|
|
|
16 |
from diffusers import FluxPipeline
|
17 |
|
18 |
torch.backends.cuda.matmul.allow_tf32 = True
|
19 |
+
huggingface_hub.login(os.getenv('HF_TOKEN'))
|
|
|
20 |
class timer:
|
21 |
def __init__(self, method_name="timed process"):
|
22 |
self.method = method_name
|