ameerazam08 commited on
Commit
b545136
Β·
verified Β·
1 Parent(s): cef54d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -4,6 +4,7 @@ from torchvision import transforms
4
  from dataclasses import dataclass
5
  import math
6
  from typing import Callable
 
7
  import spaces
8
 
9
  import torch
@@ -16,7 +17,8 @@ from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5Tokeniz
16
  from safetensors.torch import load_file
17
  dtype = torch.bfloat16
18
  from huggingface_hub import snapshot_download
19
- print(snapshot_download(repo_id="nyanko7/flux-dev-de-distill"))
 
20
  device = "cuda" if torch.cuda.is_available() else "cpu"
21
  # ---------------- Encoders ----------------
22
 
 
4
  from dataclasses import dataclass
5
  import math
6
  from typing import Callable
7
+ import os
8
  import spaces
9
 
10
  import torch
 
17
  from safetensors.torch import load_file
18
  dtype = torch.bfloat16
19
  from huggingface_hub import snapshot_download
20
+ model_path = snapshot_download(repo_id="nyanko7/flux-dev-de-distill")
21
+ print("model_list ",os.listdir(model_path))
22
  device = "cuda" if torch.cuda.is_available() else "cpu"
23
  # ---------------- Encoders ----------------
24