Spaces:
Build error
Build error
Nupur Kumari
commited on
Commit
•
5d9a480
1
Parent(s):
71da51f
custom-diffusion-space
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -12,7 +12,7 @@ import torch
|
|
| 12 |
from diffusers import StableDiffusionPipeline
|
| 13 |
sys.path.insert(0, 'custom-diffusion')
|
| 14 |
|
| 15 |
-
from
|
| 16 |
|
| 17 |
# def load_model(text_encoder, tokenizer, unet, save_path, modifier_token, freeze_model='crossattn_kv'):
|
| 18 |
# st = torch.load(save_path)
|
|
|
|
| 12 |
from diffusers import StableDiffusionPipeline
|
| 13 |
sys.path.insert(0, 'custom-diffusion')
|
| 14 |
|
| 15 |
+
from src import diffuser_training
|
| 16 |
|
| 17 |
# def load_model(text_encoder, tokenizer, unet, save_path, modifier_token, freeze_model='crossattn_kv'):
|
| 18 |
# st = torch.load(save_path)
|